'''
需求：
1 程序启动后，给用户提供查询接口，允许用户重复查股票行情信息(用到循环)
2 允许用户通过模糊查询股票名，比如输入“啤酒”, 就把所有股票名称中包含“啤酒”的信息打印出来
3 允许按股票价格、涨跌幅、换手率这几列来筛选信息，
比如输入“价格>50”则把价格大于50的股票都打印，输入“市盈率<50“，则把市盈率小于50的股票都打印，不用判断等于。

stock_data.csv
'''
import pandas as pd

stock_data = pd.read_csv('stock_data.csv')

while True :
    infos = input("Please input the infomation you want to search:")
    if '<' in infos:
        bra = infos[0:infos.index('<')]
        ket = infos[infos.index('<')+1:-1]
        a = stock_data[(stock_data[bra] < int(ket))].index.tolist()
        print(stock_data.iloc[a])
    elif '>' in infos:
        bra = infos[0:infos.index('>')]
        ket = infos[infos.index('>') + 1:-1]
        a = stock_data[(stock_data[bra] > int(ket))].index.tolist()
        print(stock_data.iloc[a])
    else:
        a = []
        for i in stock_data.名称:
            if infos in i:
                b = stock_data[(stock_data.名称 == i)].index.tolist()
                a.append(b[0])
        print(stock_data.iloc[a])
    flag = input('Do you finish? Y/N')
    if flag == 'Y' :
        break


